# -*- coding: utf-8 -*-
import logging
import os

path = './log.log'

def setup_logging(log_file=path):

    os.makedirs(os.path.dirname(log_file), exist_ok=True)
    # 获取日志记录器
    logger = logging.getLogger("Bert")
    logger.setLevel(logging.INFO)
    # 避免日志重复记录
    if not logger.handlers:
        file_handler = logging.FileHandler(log_file, encoding="utf-8")
        file_handler.setLevel(logging.INFO)
        # 创建控制台处理器
        console_handler = logging.StreamHandler()
        # 设置控制台处理器级别
        console_handler.setLevel(logging.INFO)
        formatter = logging.Formatter("%(asctime)s-%(levelname)s-%(message)s")
        # 为处理器添加格式
        file_handler.setFormatter(formatter)
        console_handler.setFormatter(formatter)
        # 为控制台处理器设置格式
        logger.addHandler(file_handler)
        # 添加控制台处理器
        logger.addHandler(console_handler)
    return logger


logger = setup_logging()
